package main.leetcode.clockin.July;

import java.util.LinkedList;

/**
 * 309. 最佳买卖股票时机含冷冻期
 *
 * <p>给定一个整数数组，其中第 i 个元素代表了第 i 天的股票价格 。​
 *
 * <p>设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票）:
 *
 * <p>你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。 卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。
 *
 * <p>示例: 输入: [1,2,3,0,2]，输出: 3，解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]
 */
public class day10 {
    public static void main(String[] args) {
        System.out.println(new day10().maxProfit(new int[] {1, 2, 3, 0, 2}));
    }

    /** 动态规划 */
    public int maxProfit1(int[] prices) {
        if (prices == null || prices.length < 2) {
            return 0;
        }
        LinkedList linkedList = new LinkedList();
        int n = prices.length;
        // dp[i][0] 第i天结束，未持有股票
        // dp[i][1] 第i天结束，持有股票
        // dp[i][2] 第i天结束，冷 冻 期
        int[][] dp = new int[n][3];
        dp[0][1] = -prices[0];
        for (int i = 1; i < n; ++i) {
            // max(前一天就没有，前一天是冷冻期)
            dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][2]);
            // max(前一天就持有，今天买入)
            dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][0] - prices[i]);
            // 前一天持有，今天卖出，进入冷冻期
            dp[i][2] = dp[i - 1][1] + prices[i];
        }
        return Math.max(dp[n - 1][0], dp[n - 1][2]);
    }

    /** 滚动变量 */
    public int maxProfit(int[] prices) {
        if (prices == null || prices.length < 2) {
            return 0;
        }
        int n = prices.length;
        int unHold = 0, hold = -prices[0], freezing = 0;
        for (int i = 1; i < n; ++i) {
            int tmp0 = unHold;
            int tmp1 = hold;
            int tmp2 = freezing;
            unHold = Math.max(tmp0, tmp2);
            hold = Math.max(tmp1, tmp0 - prices[i]);
            freezing = tmp1 + prices[i];
        }
        return Math.max(unHold, freezing);
    }

    interface c {
        public String d = "";

        public abstract void d();
    }

    abstract class s {
        public String s;

        public abstract void d();
    }
}
